import os
import torch
import numpy as np

def gen_golden_data_simple():
    dtype = np.int16
    input_shape = [4095, 511]
    # output_shape = [511, 513]

    x = np.random.uniform(1, 10, input_shape).astype(dtype)

    # 使用 np.eye 生成 identity 矩阵作为 y，并把它作为 golden
    y = np.eye(input_shape[0], input_shape[1], dtype=dtype)
    golden = y.astype(np.int16)
    
    os.system("mkdir -p input")
    os.system("mkdir -p output")
    x.astype(dtype).tofile("./input/input_x.bin")
    golden.tofile("./output/golden.bin")

if __name__ == "__main__":
    gen_golden_data_simple()

